from api.ihrm_login_api import IhrmLoginApi


class TokenUtil:
  @classmethod
  def get_token(cls):
    login_body = {"mobile": "13800000002", "password": "123456"}
    login_resp = IhrmLoginApi.ihrm_login(login_body)
    token = login_resp.json().get("data")
    return token

if __name__ == '__main__':
  print(TokenUtil.get_token())
